from django.template import Library
from django.urls import reverse
from web import models

register = Library()


@register.simple_tag
def use_space(size):
    if size >= 1024 * 1024:
        return '%2f GB' % (size / (1024 * 1024), )
    elif size >= 1024:
        return '%2f MB' % (size / 1024, )
    else:
        return '%d KB' % size
